import streamlit as st
import os
from datetime import datetime

# 设置页面标题
st.title('图片上传与处理')

# 用户选择图片并上传
uploaded_file = st.file_uploader("请选择一张图片", type=['png', 'jpg', 'jpeg'])

# 当用户点击“开始处理”按钮时
if st.button('开始处理'):
    if uploaded_file is not None:
        # 保存图片到服务器
        upload_time = datetime.now().strftime('%Y%m%d_%H%M%S')
        file_name = f"{upload_time}_{uploaded_file.name}"
        save_path = os.path.join("uploads", file_name)  # 假设有一个名为"uploads"的目录用于存放图片
        with open(save_path, "wb") as f:
            f.write(uploaded_file.getbuffer())

        # 生成下载链接（此处仅示例，实际可能需要根据服务器配置调整）
        download_link = f"http://yourserver.com/uploads/{file_name}"

        # 显示处理完成消息及下载链接
        st.success("图片处理完成！")
        st.markdown(f"您可以从[这里]({download_link})下载已处理的图片。")
    else:
        st.warning("请先上传图片。")

# 注意：你需要确保有权限写入指定目录，并且web服务器可以访问这个目录以提供下载服务。